package com.zzyl.common.ai;

import com.baidubce.qianfan.Qianfan;
import com.baidubce.qianfan.core.auth.Auth;
import com.baidubce.qianfan.model.chat.ChatResponse;

public class AIModelTest {

    public static void main(String[] args) {
        ChatResponse response = new Qianfan(Auth.TYPE_OAUTH,"l8qt6nC1Qc0R20ZuFkwR4Jla","x8UWTeR8WcGtnQbOEtTb5DQ3MthIvXJO").chatCompletion()
                .model("ERNIE-4.0-8K-Preview") // 使用model指定预置模型
                // .endpoint("completions_pro") // 也可以使用endpoint指定任意模型 (二选一)
                .addMessage("user", "如何挣到100块") // 添加用户消息 (此方法可以调用多次，以实现多轮对话的消息传递)
                .temperature(0.7) // 自定义超参数
                .maxOutputTokens(2000)
                .responseFormat("json_object")
                .execute(); // 发起请求
        System.out.println(response.getResult());
    }
}
